package 排序;

import java.util.*;

/**
 * @Author: biubiu0002
 * @Date: 2021-09-03 16:33
 **/
public class 最小K个数 {
    public static void main(String[] args) {
        int[] input = {1, 3, 5, 7, 2, 4, 6, 8};
        int k = 4;
        System.out.println(new 最小K个数().smallestK(input,k));

    }

    public int[] smallestK(int[] arr, int k) {
        /**
         * 快速排序
         */
//        int[] res = new int[k];
//        Arrays.sort(arr);
//        for (int i = 0; i < k; i++) {
//            res[i] = arr[i];
//        }
//        return res;

        /**
         * 堆排序
         */
        int[] res = new int[k];
        if(k==0) {
            return res;
        }
        Queue<Integer> queue = new PriorityQueue(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2-o1;
            }
        });
        for(int i=0;i<k;i++){
            queue.offer(arr[i]);
        }
        for(int i=k;i<arr.length;i++){
            if(queue.peek()>arr[i]){
                queue.poll();
                queue.offer(arr[i]);
            }
        }
        for(int i=0;i<k;i++){
            res[i] = queue.poll();
        }
        return res;



    }
}
